include legend in plot python